package com.smart.tool.core;

import org.apache.commons.compress.utils.IOUtils;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.nio.charset.Charset;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;

/**
 * @Author xiaoqiang
 * @Version CompressUtils.java, v 0.1 2025年05月17日 23:11 xiaoqiang
 * @Description: TODO
 */
public class CompressTool {
    public static String compress(String str, Charset charset) throws IOException {
        if (null == str || str.length() <= 0) {
            return str;
        }
        // 创建一个新的 byte 数组输出流
        ByteArrayOutputStream out = null;
        // 使用默认缓冲区大小创建新的输出流
        GZIPOutputStream gzip = null;
        try {
            out = new ByteArrayOutputStream();
            gzip = new GZIPOutputStream(out);
            // 将 b.length 个字节写入此输出流
            gzip.write(str.getBytes());
            gzip.finish();
            // 使用指定的 charsetName，通过解码字节将缓冲区内容转换为字符串
            return out.toString(charset.name());
        }finally {
            IOUtils.closeQuietly(out);
            IOUtils.closeQuietly(gzip);
        }

    }

    /**
     * 字符串的解压
     */
    public static String unCompress(String str,Charset inCharset,Charset outCharset) throws IOException {
        if (null == str || str.length() <= 0) {
            return str;
        }
        // 创建一个新的 byte 数组输出流
        ByteArrayOutputStream out = new ByteArrayOutputStream();
        // 创建一个 ByteArrayInputStream，使用 buf 作为其缓冲区数组
        ByteArrayInputStream in = new ByteArrayInputStream(str
                .getBytes(inCharset.name()));
        // 使用默认缓冲区大小创建新的输入流
        GZIPInputStream gzip = new GZIPInputStream(in);
        try {
            out = new ByteArrayOutputStream();
            in = new ByteArrayInputStream(str
                    .getBytes(inCharset.name()));
            gzip = new GZIPInputStream(in);
            IOUtils.copy(gzip, out);
            // 使用指定的 charsetName，通过解码字节将缓冲区内容转换为字符串
            return out.toString(outCharset.name());
        }finally {
            IOUtils.closeQuietly(gzip);
            IOUtils.closeQuietly(in);
            IOUtils.closeQuietly(out);
        }

    }
}
